package com.tanhua.sso.service;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.tanhua.sso.mapper.UserMapper;
import com.tanhua.sso.pojo.User;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;


import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class UserService {

    @Autowired
    private RedisTemplate<String,String> redisTemplate;

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private HuanXinService huanXinService;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    /**
     * 用户登陆
     * @param phone 手机号
     * @param code 验证码
     * @return
     */
    public String login(String phone, String code) {

        String redisKey = "CHECK_CODE_"+phone;

         boolean isNew=false;
        
        //从redis中获取验证码
        String redisData = this.redisTemplate.opsForValue().get(redisKey);

        //判断用户输入的验证码是否正确
        if (!StringUtils.equals(code, redisData)) {
            //如果验证码不正确返回null
            return null;
        }

        //验证码校验完后，需要删除
        this.redisTemplate.delete(redisKey);

        //判断手机号是否已经注册过
        QueryWrapper<User> queryWrapper = new QueryWrapper<User>();
        queryWrapper.eq("mobile",phone);
        User user = userMapper.selectOne(queryWrapper);

        if(null==user){
            //注册用户
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456"));
            this.userMapper.insert(user);

            isNew=true;

            //注册用户到环信
            this.huanXinService.register(user.getId());
        }


        Map<String, Object> claims = new HashMap<String, Object>();
        claims.put("id", user.getId());

        // 生成token
        String token = Jwts.builder()
                .setClaims(claims) //payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐
                .setExpiration(new DateTime().plusHours(24).toDate()) //设置过期时间，24小时后过期
                .compact();


        try {
            //发送用户登录成功的消息
            Map<String,Object> msg = new HashMap<>();
            msg.put("id", user.getId());
            msg.put("mobile",phone);
            msg.put("date", System.currentTimeMillis());

            this.rocketMQTemplate.convertAndSend("tanhua-sso-login", msg);
        } catch (MessagingException e) {
            log.error("发送消息失败！", e);
        }

        return token+"|"+isNew;
    }

    /**
     * 校验token
     * @param token
     * @return
     */
    public User queryUserByToken(String token) {

        try {
            // 通过token解析数据
            Map<String, Object> body = Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
            //{mobile=1333333333, id=2, exp=1605513392}

            User user = new User();
            user.setId(Long.valueOf(body.get("id").toString()));

            //需要返回user对象中的mobile,需要查询数据库获取到mobile
            //如果每次都查询数据库，必然会导致性能问题，需要对用户的手机号进行缓存操作
            //数据缓存时，需要设置过期时间，过期时间要与token的时间一致
            //如果用户修改了手机号，需要同步修改redis中的数据
            String redisKey = "TANHUA_USER_MOBILE_"+user.getId();
           if( this.redisTemplate.hasKey(redisKey)){
               user.setMobile(this.redisTemplate.opsForValue().get(redisKey));
           }else{
               //查询数据库
               User u = this.userMapper.selectById(user.getId());
               user.setMobile(u.getMobile());
               //将手机号写入到redis中
               //jwt中的过期间单位为：秒
               long timeout=Long.valueOf(body.get("exp").toString())*10000000-System.currentTimeMillis();
               this.redisTemplate.opsForValue().set(redisKey,u.getMobile(),timeout, TimeUnit.MILLISECONDS);
           }

            return user;
        } catch (ExpiredJwtException e) {
            log.info("token已过期! token="+token);
        } catch (Exception e) {
            log.error("token不合法! token="+token,e);
        }
        return null;
    }


    public boolean updatePhone(Long id, String newPhone) {
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile", newPhone);
        User user = this.userMapper.selectOne(queryWrapper);
        if (user != null) {
            // 改手机号已经存在
            return false;
        }
        user = new User();
        user.setId(id);
        user.setMobile(newPhone);
        return this.userMapper.updateById(user) > 0;
    }
}
